package com.lbc.kmp;

import java.util.Arrays;

public class KMPAlgorithm {
    //  kmp字符串匹配算法
    public static void main(String[] args) {
        String str1 = "BBC ABCDAB ABCDABCDABDE";
        String str2 = "ABCDABD";

        int[] next = kmpNext(str2);  // [0, 0, 0, 0, 1, 2, 0]
        System.out.println("next = " + Arrays.toString(next));

        int index = kmpSearch(str1, str2, next);
        System.out.println("index = " + index);
    }

    /**
     *   写出我们的kmp搜索算法
     * @param str1  源字符串
     * @param str2  子串
     * @param next  子串对应的部分匹配表
     * @return  如果是-1就是没有匹配到，否则返回第一个匹配的位置
     */
    public static int kmpSearch(String str1, String str2, int[] next) {

        // 遍厉
        for (int i = 0, j = 0; i < str1.length(); i++) {

            // 需要考虑 str1.charAt(i) != str2.charAt(j) 的情况，去调整 j 的大小
            // kmp算法的核心点，可以验证。。。
            while (j > 0 && str1.charAt(i) != str2.charAt(j)) {
                j = next[j-1];
            }

            if (str1.charAt(i) == str2.charAt(j)) {
                j++;
            }
            if (j == str2.length()) {  // 找到了
                return i - j + 1;
            }
        }
        return -1;
    }

    //  获取到一个字符串（字串）的部分匹配值  -- 最大长度相同前后缀？
    public static int[] kmpNext(String dest) {
        //  创建一个 next 数组保存部分匹配值
        int[] next = new int[dest.length()];
        next[0] = 0;  // 如果字符串的长度为1，部分匹配值就是0
        for (int i = 1, j = 0; i < dest.length(); i++) {
            // 当 dest.charAt(i) != dest.charAt(j) 时，我们需要从next[j-1]获取新的j
            // 直到我们发现 有 dest.charAt(i) == dest.charAt(j) 成立才退出
            // 这是kmp算法的一个核心点
            while (j > 0 && dest.charAt(i) != dest.charAt(j)) {
                j = next[j-1];
            }

            // 当 dest.charAt(i) == dest.charAt(j) 满足时，部分匹配值就是 +1
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }

}
